absl-py~=1.0
jax~=0.4.24
jaxlib~=0.4.24
tensorflow~=2.15
chex~=0.1.81
